import os
import shutil

# 定义目录路径
dir_path = os.path.join(os.getcwd(), "01-exercise/01-20/arrange_dir")

# 遍历目录中的所有文件
for file in os.listdir(dir_path):
    # 获取文件完整路径
    file_path = os.path.join(dir_path, file)
    
    # 跳过目录，只处理文件
    if not os.path.isfile(file_path):
        continue
    
    # 获取文件扩展名（不含点）
    ext = os.path.splitext(file)[1][1:] or "no_extension"
    
    # 创建目标目录（如果不存在）
    target_dir = os.path.join(dir_path, ext)
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)
        print(f"创建目录: {target_dir}")
    
    # 构建目标路径
    target_path = os.path.join(target_dir, file)
    
    # 移动文件
    try:
        shutil.move(file_path, target_path)
        print(f"移动文件: {file} -> {target_path}")
    except Exception as e:
        print(f"移动文件失败: {file} - {str(e)}")

print("文件整理完成！")